tensor([[[0.3600, 0.0000],
         [0.4000, 0.2300]],

        [[0.6400, 0.4400],
         [0.6000, 0.0000]],

        [[0.0000, 0.5600],
         [0.0000, 0.7700]]])